tried stuff with loops in python