*added the work I did on HW5, will finish later