Still trying to fix edit functionality.