Trying to get back on track

Extension was giving some errors. Corrected those.