Move some UI stuff over

Signed-off-by: regentzu <regent.zu@gmail.com>