Changed the program a bit