Quality Code changes to Unit of Work.