Add image flags to model and view