Trying to replace raw_input