Add argument to easily change initialization vector