Added Assert statements to test