added --trajectory option to plot a gradient-descent trajectory on top of the heatmap