Edit and remove are working