still working on correct weights