You can get better code by exploiting model weights
kelvinfichter.com