Move drawing of simulation results into prediction class itself