IPython 2.3 and IPython3 compatibility

Heatmap now works with IPython2.3 as well as IPython 3. Fixes #1.