Added Network class and setting up the GUI