Almost have output working