Update TestStep

* Implementing main function.
* Removed getArg. #5