add loading to final step