add remove functionality and add cleanup to style tag