adding elements to the ui and improving design