added small imagenet tests