add tests for vital step functionality