Changed scales package __init__ to import shape scale
Changed signature of assign_shapes to allow reference to ggplot object