Updated some gui code