Add test for error handling
More tests