Update learn.py