Modify raw_input => input