added suggested ui improvements