IamManchanda code improvement: leetcode