Added cool test case and removed print statements