Update __init__.py

[JacobKosowski]

os.remove tries to remove a single file, while the code expects the removal of a directory. os.rmdir could fail if the directory is not empty.