Adding another test file and fixing previously broken test