added image class in features.py