adding/improving GUI