Removed print statements and changed script to take command
line input instead of a hard-coded value.