some cleanup on learn.py