Updated to work with matplotlib 1.3.1
Merge pull request #2 from jandic/master

Updated to work with matplotlib 1.3.1 (thanks @jandic)