Added check on Wifi