Added ability to download models