Added visualization.
Fixed styles of UI.