we clearly want attention in Models