Added models fixed bug in generation of test and trainign set with labels