Simplify PY2 raw_input check