The research work presented in this paper is part of an innovative framework that deals with the design process of lower limb prostheses. The quality of the whole prosthesis depends on the comfort of the socket, which realizes the interface between the patient body and the mechanical parts. We developed a CAD system, named Socket Modelling Assistant that guides the user during the design of the socket, exploiting domain knowledge and design rules. In this work we present a preliminary study that describes the implementation of a software module able to automatically identify the critical areas of the residuum to adequately modify the socket model and reach the optimal shape. Once the critical areas have been identified, the Socket Modelling Assistant can apply proper geometry modifications, in order to create the load and off-load zones for a good pressure distribution over the residual limb.