Prevent index from going out of bounds

[Tayomide]

That was a mistake. I plan on adding more categories from Leetcode later today. I'll make a PR with all the changes