Fix input bug in the example code

- input to raw_input