Bug fixes, v0.1.1