Use raw_input when running in python2