Update plotting, remove printing