Exercise for Real Python 3