working on adding gui