Modifications for testing

• train_data -> modify that this script does the training (no testing)
• test_training -> modify this for test purposes
• reinitialize path for pytest
• fix bug within training!