working on unit test
working on unit test
working on unit  test and cleaning code
finished unit tests