small change in code