Logic tests added