Add schema
Fix session

Remove to_yaml