For ben's python code