UI re-layout for new functions