Fixed some of the code blocks