First unit test working