Add diagram (not finished)