Python 2/3 support for command line input