Improve UI

* more type handling
* better vertical sizing