First attempt at HW2