Add header comment and test case.
Solution for SortColors using counting sort.