Modified  the program a little bit