Added custom model template. Not sure if doing it correctly.