Fix working version of schema grapher