Added more drinks and more training data