Add remove button to modules