Finished first version of HW4