https://remp3indir.net/plot-square-using-python/