UPDATED: Added wigner_cmap to init and fixed docstrings