Plots working! Still need to clean up output tho