changed image models class name