Edited main.py, prompts for input instead of taking an argument in command line. Updated readme accordingly.