fixed plotting errors
small change