add ML saftey features