South Australian company PMB Defence will play a key role in the AUKUS submarine program, following a new contract.