diff --git a/data_structures/stacks_and_queues/1_stack_push/Solution/solution.py b/data_structures/stacks_and_queues/1_stack_push/Solution/solution.py index 5d9f715d..7c5eea1b 100644 --- a/data_structures/stacks_and_queues/1_stack_push/Solution/solution.py +++ b/data_structures/stacks_and_queues/1_stack_push/Solution/solution.py @@ -1,7 +1,12 @@ # Code your solution here -def push(stack_a): - stack_a.push('Python') - return stack_a -stack_a=['C','Perl','C++','Java'] -result=push(stack_a) -print(result) \ No newline at end of file +from typing import Any + + +class Stack(list): + def push(self, value: Any) -> Stack: + self.append(value) + return self + +stack_a = Stack(['C','Perl','C++','Java']) +result = stack_a.push('Python') +print(result)