I think the right way to go is to add some functions that misbehave.
The students would explain why the functions misbehave and rewrite them to get the correct form of these functions.