Add remove logic
Fix os.path.exists