Python Return In A With Statement