Remove import of jsonschema from schema.py, was causing import error if user didn't have module