Bug fix: Update documentation