Created class that we'll need in the gui