shrunk model size again for testing purposes