Add images for state diagram