Adding Assert tests