Added missing return
Added sub heading to output