Minor ui fixes
release 0.0.2