improve prototype gui