Use attributes instead of hard coding