Adding helper classes for unit tests