UI stuff -by Mengyuan Gu