Updated so heavy nodes get added in.