Changed directory to output/ for plots