Added start of unit tests