Got rid of raw_input()