Make argment generic