dded faster test and made trainset smaller... also changed some parameters