Improve some comments
Correct muon pt sort function