Feature/diagram widget