Use IPython directive for code documentation

... as the documentation is still young, it makes sense to set this up early on and use it consistently throughout the documentation.