Weekly Python Exercise A1: Data structures for beginners