modify js to take small,medium, and large image