Changed up th UI