Almost completed HW 4