Python Optional Return