Touch up formatting on code output