added note to predict code