update code
changed plot