Edit the Producer and Consumer class