add state diagrams