Updates UI and adds functionality to match