Added test for each epoch