Fixed the entire UI. Is nice!