Added predicted state to main