Little clean up things for learner.py