Got UI working again