Added delete button. Works properly now.