Added state transitions for display