Edit code blocks so code is better styled