work on better gui