Add get_columns to functions module main import