Update DurationLearner.py