slight improvements on ui