Bug-fixes for version 0.1.0