Python coding challenge for beginners