clean up code on program #1