Sort Color Problem Leetcode Python Easy