I need to improve my dataset but this more or less does what it's supposed to. weight set to 1.15 seems to work decent.