implemented second usecase
added unit tests
added todo