Played aroudn with regression tree