Creating a Unit of work class