write answer for wine_classify.py