Add basic protection support