Added constructor parameter to Entity class