Added edge cases tests to ClassTest