Add image model and controller tests