Add case for Edit. Uncomment line to use