GUI restoration almost done