Use right input functions in PY2/3