Initial implementation of HW2