Changing from input to argument