Use a new class to handle the upgrade check and show the output