Update lighting code