Added input = raw_input (to test)