Python exercises for technical interview