Added codes for train and test in point wise