Issue with weights and states and updating test data.