v.0.6.1 Bug fixes