Improve UI. Version 1.1