ui improvements
new unit test