Correcting my grammar
Adding columns property
Adding casted columns to columns array
Using $query->id when finding flow assignments
Adding select to columns array before we've changed the variable
Adding logic to select method