assemble_data.py fix for keyboard interrupt