Improve code view

Make some better styles