Adding a bit more logic to small images