Column Highlighting Problem Python Implementation