Added basic support for code. Needs some work.