Add some code that can possibly allow for a GUI to work out!