in-lab work

cleaned up the code a bit, added a list
started work on problem 4