Bug fixes, v 0.2.2