Adding test cases.
100% code coverage, boom