Added Unit testing to the class