Fixed small raw_input bug, if python2 input = raw_input else input.