Make it possible to pass code blocks.