Updated entity api