Second unit test working