Select the column directly
Python 3 mostly passing