UI improvements.
Added some comments.