Code Update

update the code to handle 3d models.