Import pyplot
Tweak lepton particle code
Change time limit 180 seconds and give -1 penalty on exceeding it