Updated user interface and added control button