adding initial code
changed the tests