Fixing documentation - should import api