Adding better code sections in docs