GUI needs finishing