Adding lighten and restore functions