https://remp3indir.net/different-squares-codefights-readable-code-python/