Update toggle_turtle_shape.py

Minor refactoring and testing with two turtles.