Change remove function to remove on long tap