Add predicting state to UI