Replace raw_input with six.moves.input