Why do not we put this?

var prediction here and we can remove form line 54