Added the new state diagram