Updated the entities based on feedback from class.