Add support for update function to entities.