Redesigning our output() function