Semi-working UI code now.