Added UnitTests to the project