Quick change to include IO in main.