Graph and ui mostly done