Python case study 1