Replace raw_input() with input()