Add support for editing tasks