Updated img to include class