Updated Entity API module