Edit button works