Model improvement

shape