added image visualization <still needs ipa_model wrapper>