Largest Item In Stack

largest_in_stack.py

class MaxStack(Stack):
    def __init__(self):
        Stack.__init__()
        self.max = []
    
    def push(self, item):
        self.items.append(item)
        if item > self.max):
            self.max.push(item)
            
    def pop(self):
        item = self.items.pop()
        if item == self.max.peek():
            return self.max.pop()
        return item
    
    def get_max(self):
        return self.max.peek()