Added files for custom models