Sort Colors Problem Leetcode Python Basic Programs