Govt extends cost of living assistance of up to $24.5k for trainees