Updating functionality and UI throughout the app