added line to save figure