Add schema support to PyKronos