Adding do-while loop for calculating vs quitting program