Added a GUI. Because i love JDialog