added missing tests. Still need to add in additional missing features like WithIndex, Explain, etc...