Updated source to show diff ways to writing the code (1-liner, 2-liner, & long way).