fixed the bug with new images. improved the model. needs clean up, but working.