Work on splitting up the driver into multiple files (we can ignore the .pyc files this produces)