class Coffee:
    ''' Coffee class.
        Attributes: size, price, name, ho, holiday_cup
        Size (string) is required in the __init__ function
        price (float) and hot (bool) are optional
        name (string) and holiday_cup (bool) are not taken int the __init__

        Methods: add on (like a flavor or shot), __str__ and __eq__
    '''
        
    def __init__(self, size, price = 2.49, hot = True):
        self.size = size
        self.price = price
        self.hot = hot
        self.name = 'PSL'
        self.holiday_cup = True

    def add_on(self, add):
        if add == 'shot':
            self.price += .99
        elif add == 'flavor':
            self.price += .50

    def __str__(self):
        coffee_str = self.name + "\n" + \
                    "Size: " + self.size + \
                    ". Price: ${:.2f}".format(self.price)
        return coffee_str

    def __eq__(self, other):
        return self.size == other.size and self.name == other.name