Set classnames on code figure