Adding changes for entity prediction