import time


def timeit(function):
    """
    Decorator to measure time take to execute a function
    """
    def wrapper(*args):
        """
        Wrapper definition for the function
        """
        start_time = time.process_time()
        output = function(*args)
        print(
            "Module: {} took: {}s".format(
                function.__name__,
                time.process_time() - start_time))
        return output
    return wrapper