Added new test and fixed minor bug.