Add #init_weights method
Add #init_weights method