diff --git a/load_data.py b/load_data.py
index c51aa27..00ff6d1 100644
--- a/load_data.py
+++ b/load_data.py
@@ -274,7 +274,26 @@ def get_demonstrations(self, frame_shape = (84, 84), recompute_demonstrations =
 
                         elif filename.endswith('.tar.bz2'):
                             with tarfile.open(extract_dir.name + "/" + filename, 'r') as tar_archive:
-                                tar_archive.extractall(extract_dir.name)
+                                def is_within_directory(directory, target):
+                                    
+                                    abs_directory = os.path.abspath(directory)
+                                    abs_target = os.path.abspath(target)
+                                
+                                    prefix = os.path.commonprefix([abs_directory, abs_target])
+                                    
+                                    return prefix == abs_directory
+                                
+                                def safe_extract(tar, path=".", members=None, *, numeric_owner=False):
+                                
+                                    for member in tar.getmembers():
+                                        member_path = os.path.join(path, member.name)
+                                        if not is_within_directory(path, member_path):
+                                            raise Exception("Attempted Path Traversal in Tar File")
+                                
+                                    tar.extractall(path, members, numeric_owner=numeric_owner) 
+                                    
+                                
+                                safe_extract(tar_archive, extract_dir.name)
                                 self._update_png_name_df(tar_archive.getnames())
                         print('%s has been processed.' %(filename))