Add doctests, bump to 0.2.1