Added comments and deleted unnecesary lines

in task 30 deleted lines with unnessesary tagging and made the program
simpler, devided train and test data