Adding loading for training