adding more unit tests
Handling the NaN use cases.