Output works, need to quickly fix counts