- updated more on the UI