Fixes the highlighting and adds code box