Added more code generation functionality.