tweaks to the ui and fixing a bug