This is the final version of the code before converting it into a proper pip-installable Python package.