Added a note about wasserstein_distance() to documentation + cleaned up docs for init_diagrams()

def setup(app):
    app.add_javascript('jsMath/plugins/spriteImageFonts.js')