Added Vitterbi test and fixed bug (was returning wrong variable)