Edit button does what it should now.