Working on UnitTest