Cleaned up function predict_increase_type