Comments and saving

Will now save the plot to a correctly named file.