Can set the key size through the argument