HackerRank.com / Python / Introduction: Write a function