Add: proper input parameters