Add spec for diagram