Using a python2atexit module