Weekly Python Exercise A2: Functions for beginners