Added custom model to Energy battery