[#44] added tests and conitnuing work on the tests