Display output redesigned. Now is everything in a one function. I saved some lines.