Basic example in diveintopython