Refactored unit of work classes.