Add a basic unit test for FSM. Get it to pass the first several tests