[Chapter 5] plot gradient ascent