small change the ui layout