Added more functionality for 3D models