Update as of latest Python package structure used internally

## Description

I've created this issue so as to update the repository since there are some things I've updated myself in other forks from this repository, so it would be nice to bring those changes here, since this feels already a bit out of date!