Made output plotting look nicer