* fix: rework challenge to use argument in function * fix: add solution * fix: use MathJax to improve math notation