finished - working with tests