Adding edit controller