Updated user interface

Add:)