started adding unit tests
now with tests