Tiny clean up and update to v0.1.1