add solution with python