Small UI changes for release