Systems and methods for creating a treatment plan for cosmetic procedures, for example, a hair transplantation procedure, are provided. The treatment plan allows user to provide input on a proposed hair element using a free-hand drawing in a two-dimensional plane, and for the proposed hair element to be generated and displayed on a three-dimensional model of a body surface. Various techniques and methods described in the application provide for improved planning of a natural looking hair.