Add support for INSERT and UPDATE statements