Modified Training code

Added folders for training data and weights