Unit tests homework