Edit button works correctly