Swapped out repositories and added UnitOfWork class.