Add tests around error handling
0.0.2