Python Set Time Limit On Function Correcting