Check that column exists in the dataset in the same way for all of the methods.

[nirbarazida]

Check that column exists in the dataset in the same way for all of the methods. Some times using a wrapper and sometimes using try-except.