next steps
plotting full weights