Add an update function and a method to get an entity's controller to Entity