Updated model validation notebook