FIX: Update docs