Minor changes to gui