Added unit testing to the project