https://remp3indir.net/coding-3n1-math-problem-using-python/