Added UoW and Repository pattern