Added draw functionality for models