Add functionality for eraser