Trying to get script to take an arguement