Avoid calling raw_input in python 3