#4 implement weight visualization