Move python files into a module