Cleaned it up so output almost works
Working now