working on unit tests, main module working well