Sort Colors Problem Leetcode Python Challenges