Edited code class