Improved comment in code