Minor update

Change variable names