Need to fix plotting errors.