Sort Color Problem Leetcode Python Beginner