Updated Entity API