You can do that. Or at least the feature is on PC, I don't know if they have it for mobile.
But you could try custom puzzles and change the category to defence. That really trains your prophylaxis and ability to spot checks.