Trying to fix code blocks