def u(z, theta) -> float:
    
    """ Defining utility function. 

    Args:
        z(float): parameter 
        theta(float): parameter

    Returns:
        Utility of assets

    """ 
    return (z**(1 + theta)) / (1 + theta)