Modify Controller to use UnitOfWork pattern instead of Repository