Fix gui

This code will compile in the project but still needs integrating with the new structure.