added in weights
fixed small change