Added import of os so os.environ.get('SEGMENT_IO_KEY') doesn't fail