Added random weights, but doesn't work fully yet