Added dependencies for training