Added JustMock to unit testing