removed layers from smaller architecture