Adding demo for AIS with graph models