minor button and ui changes