improve gui prototype