Add restore_gamma option