Update plot_freq_response.py to use proper axis