Try to use autosummary for the documentation.