def get_info_dataset(data, bool = True):
    """ 
    Function to give a broad overview of the dataset
    input: dataframe, bool
        bool = True (default) prints info
        bool = False only returns num and cat columns
    prints info, head, tail, describe, num value columns, and cat value columns
    returns num_var & cat_var
    """
    
    num_var = data.select_dtypes(include=['int64', 'float64']).columns
    categ_var = data.select_dtypes(include=['category', 'object']).columns
    
    if bool == True:
    
        print('Basic information\n','---------------------------------------------')
        data.info()

        print('\nBasic statistics\n','---------------------------------------------')
        print(data.describe().T)

        print('\nNumerical variables are:\n', num_var)
        print('-------------------------------------------------')

        print('Categorical variables are:\n', categ_var)
        print('-------------------------------------------------') 
    
    return num_var,categ_var