Updated Unit Tests for the State Chart