add test functionality