Update UI

Väiksed parandused