Add option to run on wifi only