Small update

Added a few fixes for the 0.2.0 release.