Fixes and improvements that can be done after the initial release