[TEST] working on unit testing