added some extra attributes to prediction