writing nested loops in python