Added the correct unit tests to the folder