changed the program a bit