Updated State Chart to code traceability