Added basic UI integration