Added edit to each controller