modified the UI a little