Replace flash id with class