Fix bugs & update documentation