Added weight plotting functionality