Code for problem "Sort Colors".