I kept this as it is. It was failing a test in which sum of the weights equals length of the class.