Last part of HW4 just before deadline