Need to update figure environment