add option to run either for electron or muon channe