enhance main ui