HW4 done but not commented etc