Improved code
Add a comment