Add solution in python