Multiple Return Statements Python