Completed Problem 4 except for comments

I'll add them in tomorrow.