Add a flag to inspect prediction scores