moving code towards .py scripts and modules instead of a notebook