Add in extension correctly