Add test for #2
Release 0.2.1