Add solution 1 in python
Add solution 2 in python