Added tests for step functionality