Added code to perform posterior inference on missing unit types