assign_loc now assigns locations