Trying to fix our plot