minor fixes needed to match the provided code.