Added support for adding different code types