Added correct variable names to comments