Small change to gui