Created a seperate class for GUI