Feat kron

Added support for higher rank pdes