HW 4

With tests graded