Math is correct for part 1 of the assignment. Also created a test program for while loops