fix some of the code blocks