https://remp3indir.net/conditional-function-in-python/