Updated entity module