Add support for updating a plan