TODO update; minor UI bugfix.