Update for v0.2.1 release