Adding initial state diagram