Update the docs to new theme

Based on astropy theme