Add code tag support