Sort Colors Problem Leetcode Python Practice