Made changes to main method to use built-in input() method from Python 3 to read command-line input