Added testing logic.