Edit button updates... still not working