Adding wifi support