Also update documentation