Toggle edit mode working... but I want to find a better way to do it.