Edit button works properly