Next to add: prediction stage (scene3)