adding support for key type args