Added a testing class and added asserts.