hw2 - fix 1a + tests