Fixed the bug and added a new test file!