I changed the program a bit