Context Manager Python Interview Question