Only add the toolbar to code and markdown cells
Make extension a little more efficient