Added Assert statements to test
Added Assert statements to test