Improve look of code blocks and general design