Add more unit tests
0.2.1 release!