improvements

small improvements on UI