Fixed return bug in spatial solver, added back returns, added input dictionary key pairs to output dictionary and updated docs