Updated code in exercise file.

The notation was different from the code in 02-fundametals.md.

Fixes #525.