I made that change, and added a separate button to delete the layer :)