Writting unit tests