Added simple testing setup