Weekly exercise 2: Tax function