update the main src; added a new test case