Small updates/bugfixes