string manipulation and while loops in python