some ui improvements. hopefully