add weights improvement idea