Fix UI

Functionality still not working