Avoid Nesting Loops in Python