You can get better code by exploiting model weights

kelvinfichter.com