Improve assertion examples
Change variable name in the square example