Added more tests.
* Will add tests for float input soon as well.