Basic Unit testing has been added