working on unit tests
fixing tests