Converting a script to a function