Writting unit test