Minor Changes to UI