Add code class support