Modified script to plot either tke contours or w contours