Small change
Improved code