Improves method name and comments to insert_prediction