Added print statements and modified assertion tests