Add wifi connection control