Add support for parsing PARTITION BY KEY clauses