Adding import statement of atexit to prevent it crapping out