Test immediately after training (should've written this weeks ago..)