Add wifi support