Added more plotting stuff based on linkcomm demo.