refactor: Move python cleanup code into own go file