Python Data Engineer Coding Question