Added unit testing setup