Write unit tests (40%)