Enhancement: add test setup