Fix Pep8 Fail
Fix bug with column selection
Fix expected text in tests