Writing a Python module for a submission script