Added class for what will become GUI