Currently, you only have a rectangle for the states. it's be nice to be able to select a cylinder (horizontal), diamond, star and maybe some other shapes. Also, be able to set those to certain colors as a prefab. That way, whenever I have a state that uses a 'int switch', I can click on the red diamond and whenever I have a series of states to start, then I can click on the 'blue cylinder'.
Now, they don't have to be perfect diamonds or cylinders. They can still be rectangles, but with the top/bottom bent. That would save screen space.