#63 fix UI