HW2: Made all corrections to HW2