added a rough diagram editor