Reworked code to use raw_input.