Add diagram to solution