if/else to allow the done notification to be displayed