Rename method assign_roles_to_nodes
Fix arguments of method