better formatting of code class, hopefully get code