Updated extension to use CodeInspection