Added testing setup