HW4 almost finished