Added plotting stuff to wifi tests