Added input and print statements