some UI touch ups