Added tests for workflow