Made this script take arguments