plot error

Hi, your work is greatly helpful. But I found a small bug. When I tried to plot the heat map of phase, the title, cbar y_label and y_label is fixed to "Amplitude". Hope a quick fix. Best wishes. :)