Added function to visualize collision model