Started working on adding a GUI