Removed comments and fixed test case.