Working training (it seems). Added test file