Minor bugfixes/update