Script now restores wifi at end, formatting