added init of weights