adding safe training module and direct import