First draft of hw2 rotten_potatoes