Trailytics AI interview question

python very very basic stuff