Simpler with raw_input.