Make changes to Unit testing files