Cleaned up plot function of extraneous comments