There are various print messages scattered throughout the code, it would be nice to include a proper logger