commit to merge
merge
fix vizualiztion bug when prediction is off the screen