There is quite a lot of html processing logic in the comment class, it might be better to create a new code renderer subtype and store it all there instead.

Please note that at present there is no way for theme designers to alter the UI of comments.