Quick changes to add user interface