add sol in python