funkcja remove_and_predict