Return element

I have a use case where I need to remove the inserted styles at a later point. What do you think about returning the inserted `elem`?