Initial unit test works