Fixed small raw_input bug. If python2 input = raw_input else input.