Improve code view

Now it's more similar to GitHub