Update notes for submitted changes

* Move docstring inside relevant class