Add error handling
Fix test
v0.0.4