started working on tests
added test