learn.ipynb - cleaned up the mapping code a bit