Guessing now works on training problems of certain sizes