Updated for docs changes