Add attr function. TODO: add test