added code for training/testing
remove reltaed files