Replaced IdentityMatrix field in utils with a function that generates
identity matrices based on given sizes.
Updated doc.