Update schema to reflect import fix
Add docstrings to schema