Seems like a <10 line change to the schema file and that conditional block there if someone wants to send a PR. I am happy to help you :)