Major updats to UI