Improve assert example