Python Program Return Value Optimization