change main program to get data by raw_input