Add more edit functionality