added GUI (better)