Checking python version to use the correct input function in __main__