Added convenience methods to UnitOfWork

These will probably get refactored out later