Finalize UI functionality