ui

Added some nice UI features.