Added testing feature for vitterbi