still trying to fix predicition