Update Entity class