Added plotting functionality for verification