Add a base class for unit tests