Added balance printing to example