Better layout for the documentation. Closes #374