some changes for better model comparison, models have same output still