Updated entity API