Works in Python 3.x now

Added figure labels, etc.