HW0 part 3 - classes