Add K Lambda specs for assignment primitives

And add type checking for the first argument of set and value