Improve maximum depth solution.
Add sort color solution