Fix bugs, update documentation