Added unit test task