changed visualization to include new data