converting *args method into a **kwargs method to handle the weights in a better way