Add wifi function