fixed some minor import issues

added the image logic