updated function calls in train_weights_2_layer and actually added to commit